package com.suifeng.mq.stock.config;

import org.springframework.amqp.core.Message;
import org.springframework.amqp.rabbit.connection.CorrelationData;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.amqp.support.converter.Jackson2JsonMessageConverter;
import org.springframework.amqp.support.converter.MessageConverter;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import javax.annotation.PostConstruct;

@Configuration
public class MyRabbitConfig {

    @Autowired
    private RabbitTemplate rabbitTemplate;

    /**
     * 使用json序列化机制，进行消息转换
     * @return
     */
    @Bean
    public MessageConverter messageConverter(){

        return new Jackson2JsonMessageConverter();
    }


    /**
     * 定制RabbitTemplate
     * 1.服务器收到消息就回调
     *      1).spring.rabbitmq.publisher-confirms=true
     *      2).设置确认回调
     * 2.消息正确抵达队列进行回调
     *      1)spring.rabbitmq.publisher-returns=true
     *       spring.rabbitmq.template.mandatory=true
     *      2)设置消息抵达队列的确认回调
     *
     * 3.消费端确认(保证每个消息被正确消费，此时才可以让Broker删除这个消息)
     *          手动签收消息设置：
     *          spring.rabbitmq.listener.simple.acknowledge-mode=manual
     */
    @PostConstruct
    public void initRabbitTemplate() {

        //设置确认回调
        rabbitTemplate.setConfirmCallback(new RabbitTemplate.ConfirmCallback() {
            /**
             * @param correlationData 当前消息的唯一关联数据（这个是消息的唯一Id）
             * @param ack             消息是否成功收到
             * @param cause           失败的原因
             */
            @Override
            public void confirm(CorrelationData correlationData, boolean ack, String cause) {
                /**
                 * 1.做好消息确认机制(publisher,consumer【手动ack】)
                 * 2.每一个发送的消息都在数据库做好记录。定期将失败的消息再次发送
                 */
                //服务器收到了
//                String id = correlationData.getId();
                //TODO 根据id修改消息的状态 1-已发送
                System.out.println("confirm...correlationData[" + correlationData + "]==>ack[" + ack + "]==>[" + cause + "] ");
            }
        });


        //设置消息抵达队列的确认回调
        rabbitTemplate.setReturnCallback(new RabbitTemplate.ReturnCallback() {

            /**
             * 只要消息没有投递给指定的队列，就触发这个失败回调
             * @param message   投递失败的消息详细信息
             * @param replyCode 回复的状态码
             * @param replyText 回复的文本内容
             * @param exchange  当时这个消息发给哪个交换机
             * @param routingKey 当时这个消息用哪个路由键
             */
            @Override
            public void returnedMessage(Message message, int replyCode, String replyText, String exchange, String routingKey) {
                //TODO 报错误了。修改数据库当前消息的状态
//                String id = message.getMessageProperties().getHeaders().get("spring_returned_message_correlation").toString();
                System.out.println("Fail Message["+message+"]==>replyCode["+replyCode+"]==>replyText["+replyText+"]==>exchange["+exchange+"]==>routingKey ["+routingKey+"]");
            }
        });
    }



}
